from k_mean import k_mean


if __name__ == '__main__':
    samples = [
        [0, 0], [1, 0], [0, 1], [1, 1],
        [2, 1], [1, 2], [2, 2], [3, 2],
        [6, 6], [7, 6], [8, 6], [7, 7],
        [8, 7], [9, 7], [7, 8], [8, 8],
        [9, 8], [8, 9], [9, 9]
    ]
    k = 2
    centers, sample_to_class, iter_count = k_mean(samples, k, 0.01)
    print('centers: ' + str(centers))
    print('classes: ' + str(sample_to_class))
    print('iters: ' + str(iter_count))
